package com.ysz.redisson.config;

import org.springframework.aop.framework.AopContext;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.AbstractBeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;

import java.util.LinkedHashMap;

/**
 * @ClassName SpringUtils
 * @Description 拿到spring的applicationContext容器以及spring的上下文环境
 * @Data 14:55
 * @Version 1.0
 * @Author ysz
 */
@Component
public final class SpringUtils implements ApplicationContextAware, BeanFactoryPostProcessor {

    /**
     * 可以拿到spring容器中的对象  spring 上下文
     */
    private static ApplicationContext applicationContext;
    /**
     * 在bean被spring实例化之前可以修改bean
     */
    private static ConfigurableListableBeanFactory configurableListableBeanFactory;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        SpringUtils.applicationContext = applicationContext;
    }

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
        SpringUtils.configurableListableBeanFactory = configurableListableBeanFactory;
    }

    /**
     * 根据bean的名字获取bean对象
     *
     * @param beanName
     * @return
     * @throws 当要获取的bean不存在抛出NoSuchBeanDefinitionException
     * @author ysz
     */
    @SuppressWarnings("all")
    public static <T> T getBeanByName(String beanName) {
        T bean = (T) SpringUtils.applicationContext.getBean(beanName);
        return bean;
    }

    /**
     * 根据class对象拿到bean对象
     * 注意，此方法将会返回传入的class对象类型的bean对象以及该bean的子类对象
     * 以linkedHashMap返回，key为bean的名字，value为bean实例
     *
     * @param clazz
     * @return
     * @throws
     * @author ysz
     */
    @SuppressWarnings("all")
    public static <T> LinkedHashMap<String, T> getBeanByClass(Class<T> clazz) {
        return (LinkedHashMap<String, T>) SpringUtils.applicationContext.getBeansOfType(clazz);
    }

    /**
     * 判断spring容器中是否包含该名字的bean
     *
     * @param beanName bean的名字
     * @return true  包含
     * @author ysz
     */
    public static boolean isContainsBean(String beanName) {
        return SpringUtils.applicationContext.containsBean(beanName);
    }

    /**
     * 返回这个bean在spring容器中的别名
     *
     * @param beanName bean的名字
     * @return 别名数组
     * @author ysz
     */
    public static String[] getAliases(String beanName) {
        return SpringUtils.applicationContext.getAliases(beanName);
    }

    /**
     * 返回这个bean的类型
     *
     * @param beanName bean在spring中的名称
     * @return 对象类型
     * @author ysz
     */
    public static Class<?> getBeanType(String beanName) {
        return SpringUtils.applicationContext.getType(beanName);
    }

    /**
     * 返回这个对象是不是一个单利的
     *
     * @param beanName bean在spring中的名称
     * @return true 是单利的
     * @author ysz
     */
    public static boolean isSingleton(String beanName) {
        return SpringUtils.applicationContext.isSingleton(beanName);
    }


    /**
     * 根据class对象拿该实例的名字
     * 此方法会把该类型,以及该类型的子类型的bean名称全部拿到
     *
     * @param beanClz bean实例的class对象
     * @return 类型对应的对象名字
     * @author ysz
     */
    public static String[] getBeanName(Class beanClz) {
        return SpringUtils.applicationContext.getBeanNamesForType(beanClz);
    }


    /**
     * 返回代理对象
     *
     * @param invoker
     * @return
     * @throws
     * @author ysz
     */
    @SuppressWarnings("all")
    public static <T> T getAopProxyObject(T invoker) {
        //内部维护了一个ThreadLocal可以拿到代理对象
        return (T) AopContext.currentProxy();
    }


    /**
     * 从spring容器中移除bean  common中已经定义好了各种对象
     * 如果在子模块中想单独定制，则需要移除spring容器中该类型对象再手动添加
     * 添加容器，请调用下面的addBean2SpringApplication方法
     * 或者子模块定义同类型不同名字的实例，使用名称注入
     *
     * @param beanName 对象名
     * @author ysz
     */
    public static void removeBeanFromSpringApplication(String beanName) {
        if (SpringUtils.isContainsBean(beanName)) {
            BeanDefinitionRegistry beanDefinitionRegistry =
                    (BeanDefinitionRegistry) SpringUtils.configurableListableBeanFactory;
            beanDefinitionRegistry.removeBeanDefinition(beanName);
        }
    }


    /**
     * 从spring容器中移除bean  common中已经定义好了各种对象
     * 如果在子模块中想单独定制，则需要移除spring容器中该类型对象再手动添加
     * 添加容器，请调用下面的addBean2SpringApplication方法
     * 或者子模块定义同类型不同名字的实例，使用名称注入
     * <p>
     * 该方法慎用，因为此方法会把该类型,以及该类型的子类型的bean名称全部拿到
     * 然后遍历删除
     * 举个例子：假设你想删除redisTemplate，当把redisTemplate的class对象传入后
     * 他还会删除该类型的子类型StringRedisTemplate在spring容器中的实例
     * 建议是根据beanName进行删除
     *
     * @param beanClz class对象
     * @author ysz
     */
    public static void removeBeanFromSpringApplicationAll(Class beanClz) {
        String[] beanNames = SpringUtils.getBeanName(beanClz);
        for (String beanName : beanNames) {
            SpringUtils.removeBeanFromSpringApplication(beanName);
        }
    }


    /**
     * 向spring容器中添加对象，方便用户在子模块中自定义配置
     * 添加后，建议使用名称注入
     *
     * @param beanClz  class对象
     * @param beanName bean名字
     * @author ysz
     */
    public static void addBean2SpringApplication(String beanName, Class beanClz) {
        BeanDefinitionRegistry beanDefinitionRegistry =
                (BeanDefinitionRegistry) SpringUtils.configurableListableBeanFactory;
        BeanDefinitionBuilder beanDefinitionBuilder =
                BeanDefinitionBuilder.genericBeanDefinition(beanClz);
        AbstractBeanDefinition beanDefinition = beanDefinitionBuilder.getBeanDefinition();
        if (!SpringUtils.isContainsBean(beanName)) {
            beanDefinitionRegistry.registerBeanDefinition(beanName, beanDefinition);
        }
    }


    /**
     * 根据key从配置文件中拿到value
     *
     * @param key 配置文件中配置的一些键值对，比如firms:A:zs 根据firms.A就可以得到zs
     *            但是注意，没法根据firms得到A
     * @author ysz
     */
    public static String getValueFromApplication(String key) {
        Environment environment = SpringUtils.getBeanByName("environment");
        String result = environment.getProperty(key);
        result = result == null ? "" : result;
        return result;
    }
}
